import torch
import torch.nn as nn
from layers import *
from conv_lstm import CLSTM_cell

#5*5的卷积
def conv5_5_relu(in_channels,out_channels,stride):
    return conv(in_channels,out_channels,5,stride,2,activation_fn=partial(nn.ReLU,inplace=True))
def deconv5_5_relu(in_channels,out_channels,stride,output_padding):
    return deconv(in_channels,out_channels,5,stride,output_padding=output_padding,activation_fn=partial(nn.ReLU,inplace=True))
def resblock(in_channels):
    return BasicBlock(in_channels,out_channels==in_channels,kernel_size=5,stride=1,use_batchnorm=False,activation_fn=)